package testDay;

import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;

public class day3_1 {
    /*
     * 1.因为三数之和固定为0，target=0-nums，把三个数之和为零变为求两数之和
     * 2.数组排序sort()，对有序数组for循环，令target=0-nums
     * 3.设置指针left，right从两端开始遍历，判断left，right指向元素的和是否等于target，
     *   若相等，res记录i，left，right三个数的值，left左移，right右移，当left>=right时退出while循环
     * 4.去重：当指针移动时，若下一个数与当前数相等，则跳过
     * 5.若nums的长度小于3，res返回null，否则返回res
     * */
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new LinkedList<>();
        if (nums.length < 3) {
            return res;
        }
        Arrays.sort(nums);
        for (int i = 0; i < nums.length; i++) {
            //去重
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            int target = 0 - nums[i];
            //  System.out.print("target " + target);
            int left = i + 1;
            int right = nums.length - 1;
            while (left < right) {
                // System.out.print("left " + nums[left] + "right " + nums[right]);
                int sum = nums[left] + nums[right];
                // System.out.print("sum " + sum);
                // System.out.println();
                if (sum == target) {
                    res.add(Arrays.asList(nums[i], nums[left], nums[right]));
                    //去重
                    while (left < right) {
                        // 不管前后相不相等，left 都要往前走
                        left++;
                        if (nums[left - 1] != nums[left]) break;
                    }
                    while (left < right) {
                        // 不管前后相不相等，right 都要往后走
                        right--;
                        if (nums[right + 1] != nums[right]) break;
                    }
                    //    ans.add(List.of(nums[i], nums[left], nums[right]));
                }
                if (sum < target) {
                    left++;
                }
                if (sum > target) {
                    right--;
                }
            }
        }
        return res;
    }
}
